from rest_framework.pagination import PageNumberPagination
from django.core.paginator import Paginator
from util.model_to_iterable import model_to_list

'1.1、drf和django都可以使用的分页器，传递的数据：模型对象集合/列表套字典数据'
class AllGenericPagination:
    '''
    web查询每日看护记录使用的分页器：
    1、使用的django的分页器
    2、传递的数据可以是数据集，也可以是列表套字典[{'id':,'name':},{}]
    '''
    page_query_params = 'page' #地址栏传递page，页码
    page_size_query_params = 'page_size'#地址栏传递page_size,页面大小
    page_size = 5 #默认的页面大小

    def __init__(self, object_list,request):
        self.object_list = object_list #数据集
        self.request = request #当前请求对象
        self.current_page = None
        self.page_size = None

    def get_page(self):
        cls = AllGenericPagination
        try:
            page = self.request.GET(cls.page_query_params)
        except Exception as _:
            page = self.request.query_params.get(cls.page_query_params)
        #1、检查page的参数是否有问题
        if not page:
            #没有携带page参数
            return f'没有携带{cls.page_query_params}参数'
        else:
            try:
                page = int(page)
            except Exception as _:
                #携带的参数不是数值
                return f'携带参数{cls.page_query_params}不是数值字符串'
        #2、检查page是否大于最大的页数
        page_tuple =divmod(len(self.object_list),self.get_page_size())
        all_page = page_tuple[0]
        if page_tuple[1]:
            all_page+=1
        if page > all_page:
            #查询的页码，超过最大页数了
            return f'查询的页码数{page}超过最大页码数{all_page}'
        #3、将当前页面数赋值给
        self.current_page = page
        return page

    def get_page_size(self)->int:
        cls = AllGenericPagination
        try:
            page_size = self.request.GET(cls.page_size_query_params)
        except Exception as _:
            page_size = self.request.query_params.get(cls.page_size_query_params)
        if not page_size:
            self.page_size = cls.page_size
        else:
            try:
                self.page_size = int(page_size)
            except Exception as _:
                self.page_size = cls.page_size
        return self.page_size

    def start(self):
        #1、检查page参数是否正常
        current_page = self.get_page()
        if type(current_page)==str:
            return {'code':400,'error':current_page}
        #2、获取到页面大小
        page_size = self.get_page_size()
        paginator = Paginator(object_list=self.object_list,per_page=page_size, orphans=0,allow_empty_first_page=True)
        # print(paginator.per_page,'每页大小')

        #3、拿到当前页结果集
        page_data = paginator.get_page(current_page)

        #拿到最终的数据集
        res = page_data.object_list
        try:
            #all 或 filter结尾的查询数据集转成列表
            res = model_to_list(res)
        except Exception as e:
            #values结尾的查询数据集转成列表
            res = list(res)
        #4、响应数据处理
        #当前页是否有下一页，上一页
        has_next = 0
        has_prev = 0
       #有下一页：当前页面大于等于1，当前页*页面大小 < 总数量
        if current_page>=1 and current_page*paginator.per_page < paginator.count:
            has_next =1
        #有上一页：当前页大于1，且总数据量要大于页面大小
        if current_page>1 and paginator.count>paginator.per_page: #有上一页，页数大于1，page_size小于总数据量
            has_prev =1

        dic ={
            'code':200,
            'msg':'获取数据成功',
            'data':res, #当前页的数据
            'current_page':current_page, #当前页数
            'count':paginator.count, #总数据量
            'page_size':paginator.per_page, #每页数据量
            'pages':paginator.num_pages, #总页数
            'next':has_next, #是否有下一页
            'previous':has_prev #是否有上一页
        }
        return dic